from flask import jsonify, render_template

def register_error_handlers(app):
    # 自定义 401 错误处理
    @app.errorhandler(401)
    def unauthorized(e):
        return jsonify(error="未授权，请先登录"), 401

    # 自定义 404 错误处理
    @app.errorhandler(404)
    def resource_not_found(e):
        return render_template('error.html', error_code=404, message="页面未找到"), 404

    # 自定义 500 错误处理
    @app.errorhandler(500)
    def internal_server_error(e):
        return jsonify(error="服务器内部错误"), 500